package com.leinao.util;

import com.jcraft.jsch.ChannelExec;
import com.jcraft.jsch.JSch;
import com.jcraft.jsch.JSchException;
import com.jcraft.jsch.Session;
import org.apache.commons.io.IOUtils;

import java.io.IOException;
import java.io.InputStream;

public class SSHUtils {
    private static final String ENCODING = "UTF-8";
    private static final int timeout = 60 * 60 * 1000;
    private static final int defaultPort = 22;

    /**
     * 远程执行命令
     *
     * @param ip 服务器IP
     * @param user 用户名
     * @param pwd 用户密码
     * @param command 需要执行的命令
     */
    public static String execCommandByJSch(String ip, String user, String pwd, String command) throws IOException, JSchException {
        return execCommandByJSch(ip, defaultPort, user, pwd, timeout, command, ENCODING);
    }

    /**
     * 远程执行命令
     *
     * @param ip 服务器IP
     * @param port 服务器端口
     * @param user 用户名
     * @param pwd 用户密码
     * @param command 需要执行的命令
     */
    public static String execCommandByJSch(String ip, int port, String user, String pwd, String command) throws IOException, JSchException {
        return execCommandByJSch(ip, port, user, pwd, timeout, command, ENCODING);
    }

    /**
     * 远程执行命令
     *
     * @param ip 服务器IP
     * @param port 服务器端口
     * @param user 用户名
     * @param pwd 用户密码
     * @param timeout 超时时间
     * @param command 需要执行的命令
     * @param resultEncoding 编码结果
     */
    public static String execCommandByJSch(String ip, int port, String user, String pwd, int timeout, String command, String resultEncoding) throws IOException, JSchException {
        Session session = getJSchSession(ip, port, user, pwd, timeout);
        String result = execCommandByJSch(session, command, resultEncoding);
        session.disconnect();
        return result;
    }

    /**
     * 远程执行命令
     *
     * @param ip 服务器IP
     * @param port 服务器端口
     * @param user 用户名
     * @param pwd 用户密码
     * @param timeout 超时时间
     */
    public static Session getJSchSession(String ip, int port, String user, String pwd, int timeout) throws JSchException {
        JSch jsch = new JSch();
        Session session = jsch.getSession(user, ip, port);
        session.setPassword(pwd);
        session.setConfig("StrictHostKeyChecking", "no"); // 第一次访问服务器时不用输入yes
        session.setTimeout(timeout);
        session.connect();
        return session;
    }

    /**
     * 远程执行命令
     *
     * @param session 会话
     * @param command 需要执行的命令
     */
    public static String execCommandByJSch(Session session, String command) throws IOException, JSchException {
        return execCommandByJSch(session, command, ENCODING);
    }

    /**
     * 远程执行命令
     *
     * @param session 会话
     * @param command 需要执行的命令
     * @param resultEncoding 编码结果
     */
    public static String execCommandByJSch(Session session, String command, String resultEncoding) throws IOException, JSchException {
        ChannelExec channelExec = (ChannelExec) session.openChannel("exec");
        InputStream in = channelExec.getInputStream();
        channelExec.setCommand(command);
        channelExec.setErrStream(System.err);
        channelExec.connect();
        String result = IOUtils.toString(in, resultEncoding);
        channelExec.disconnect();
        return result;
    }
}
